package com.zcily.demo.mq;

import com.rabbitmq.client.Channel;
import com.zcily.demo.config.RabbitConfig;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.amqp.rabbit.listener.MessageListenerContainer;
import org.springframework.amqp.rabbit.listener.RabbitListenerEndpointRegistry;
import org.springframework.amqp.support.AmqpHeaders;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.messaging.handler.annotation.Header;
import org.springframework.stereotype.Component;

@Slf4j
@Component
public class RabbitConsumer {

    @Autowired
    private RabbitListenerEndpointRegistry endpointRegistry;

    public void startRabbitMQ() {
        MessageListenerContainer container = endpointRegistry.getListenerContainer(RabbitConfig.COMMON_LISTENER);
        if (!container.isRunning()) {
            container.start();
        }
    }

    @RabbitListener(
            id = RabbitConfig.COMMON_LISTENER,
            queues = RabbitConfig.COMMON_QUEUE,
            autoStartup = "false",
            ackMode = "MANUAL"
    )
    public void receiveQueue(Message message, Channel channel, @Header(AmqpHeaders.DELIVERY_TAG) long tag) {
        try {
            // 处理消息
            String msg = new String(message.getBody());
            log.info("收到普通队列的消息：{}", msg);
            if (Math.random() >= 0.5) {
                throw new RuntimeException("随机错误");
            }
            // 确认消息
            channel.basicAck(tag, false);
        } catch (Exception e) {
            log.error("RabbitMQ Consume Error", e);
            // 否定确认，并重新入队（如果需要）
            try {
                // consumer处理失败后，例如：有值为1、2、3、4 deliveryTag的投递。
                // 1、如果channel.basicNack(4, true, true);表示deliveryTag=4之前未确认的消息都处理失败且将这些消息重新放回队列中。
                // 2、如果channel.basicNack(4, true, false);表示deliveryTag=4之前未确认的消息都处理失败且将这些消息直接丢弃。
                // 3、如果channel.basicNack(4, false, true);表示deliveryTag=4的消息处理失败且将该消息重新放回队列。
                // 4、如果channel.basicNack(4, false, false);表示deliveryTag=4的消息处理失败且将该消息直接丢弃。
                channel.basicNack(tag, false, true);

                // 相比channel.basicNack，除了没有multiple批量确认机制之外，其他语义完全一样。
                // 1、如果channel.basicReject(4, true);表示deliveryTag=4的消息处理失败且将该消息重新放回队列。
                // 2、如果channel.basicReject(4, false);表示deliveryTag=4的消息处理失败且将该消息直接丢弃。
                // channel.basicReject(tag, false);
            } catch (Exception ex) {
                log.error("RabbitMQ Requeue Error", ex);
            }
        }
    }
}
